package com.example.srb.service2oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.CannedAccessControlList;
import com.example.srb.service2oss.service.FileService;
import com.example.srb.service2oss.util.OssProperties;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;

import java.io.InputStream;
import java.util.UUID;

@Service
public class FileServiceImpl implements FileService {
    @Override
    public String upload(InputStream inputStream, String module, String originalFilename) {

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID,
                OssProperties.KEY_SECRET);

        if (!ossClient.doesBucketExist(OssProperties.BUCKET_NAME)) {
            ossClient.createBucket(OssProperties.BUCKET_NAME);
            ossClient.setBucketAcl(OssProperties.BUCKET_NAME, CannedAccessControlList.PublicRead);
        }


        // 填写本地文件的完整路径。如果未指定本地路径，则默认从示例程序所属项目对应本地路径中上传文件流。
        // 上传文件流。
        // 文件目录结构 "avatar/2021/02/27/uuid.jpg"
        //构建日期路径
        String timeFolder = new DateTime().toString("/yyyy/MM/dd/");

        originalFilename = UUID.randomUUID().toString() +
                originalFilename.substring(originalFilename.lastIndexOf("."));
        String path = module + timeFolder + originalFilename;

        // 依次填写Bucket名称（例如examplebucket）和Object完整路径（例如exampledir/exampleobject.txt）。Object完整路径中不能包含Bucket名称。
        ossClient.putObject(OssProperties.BUCKET_NAME, path, inputStream);

        // 关闭OSSClient。
        ossClient.shutdown();

        // https://srb2study.oss-cn-beijing.aliyuncs.com/test/%E5%8A%A8%E6%80%81%E4%BB%A3%E7%90%86.png
        return "https://"
                + OssProperties.BUCKET_NAME
                + "."
                + OssProperties.ENDPOINT
                + "/"
                + path;
    }

    @Override
    public void remove(String url) {
        // https://srb2study.oss-cn-beijing.aliyuncs.com/  主机host
        // test/%E5%8A%A8%E6%80%81%E4%BB%A3%E7%90%86.png  需要删除的文件路径
        String host = "https://"+OssProperties.BUCKET_NAME+"."+OssProperties.ENDPOINT+"/";

        // 填写文件完整路径。文件完整路径中不能包含Bucket名称。
        String objectName = url.substring(host.length());

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID,
                OssProperties.KEY_SECRET);

        // 删除文件或目录。如果要删除目录，目录必须为空。
        ossClient.deleteObject(OssProperties.BUCKET_NAME, objectName);

        // 关闭OSSClient。
        ossClient.shutdown();

    }
}
